pyspark flatten nested json